from Pattern import *
from common import *

class Environment:
    def __init__(self):
        self.pattern = CreatePatternFromIds([0, 2, 3, 4, 1, 3, 2, 4])

    def statesInCycle(self):
        return self.pattern.states
        
    def getState(self, n):
        n = n % len(self.pattern.states)
        return self.pattern.states[n]
    
    def size(self):
        return len(self.pattern.states)